1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
| Unknown: Fail to find the dnn implementation.
Traceback (most recent call last):
File "train.py", line 33, in <module>
num_epochs=20)
File "/home/bnu/venvs/deeplearning3.6/lib/python3.6/site-packages/textgenrnn/textgenrnn.py", line 320, in train_from_file
texts, context_labels=context_labels, **kwargs)
File "/home/bnu/venvs/deeplearning3.6/lib/python3.6/site-packages/textgenrnn/textgenrnn.py", line 291, in train_new_model
**kwargs)
File "/home/bnu/venvs/deeplearning3.6/lib/python3.6/site-packages/textgenrnn/textgenrnn.py", line 221, in train_on_texts
validation_steps=val_steps
File "/home/bnu/venvs/deeplearning3.6/lib/python3.6/site-packages/keras/legacy/interfaces.py", line 91, in wrapper
return func(*args, **kwargs)
File "/home/bnu/venvs/deeplearning3.6/lib/python3.6/site-packages/keras/engine/training.py", line 1415, in fit_generator
initial_epoch=initial_epoch)
File "/home/bnu/venvs/deeplearning3.6/lib/python3.6/site-packages/keras/engine/training_generator.py", line 213, in fit_generator
class_weight=class_weight)
File "/home/bnu/venvs/deeplearning3.6/lib/python3.6/site-packages/keras/engine/training.py", line 1215, in train_on_batch
outputs = self.train_function(ins)
File "/home/bnu/venvs/deeplearning3.6/lib/python3.6/site-packages/keras/backend/tensorflow_backend.py", line 2666, in __call__
return self._call(inputs)
File "/home/bnu/venvs/deeplearning3.6/lib/python3.6/site-packages/keras/backend/tensorflow_backend.py", line 2636, in _call
fetched = self._callable_fn(*array_vals)
File "/home/bnu/venvs/deeplearning3.6/lib/python3.6/site-packages/tensorflow/python/client/session.py", line 1454, in __call__
self._session._session, self._handle, args, status, None)
File "/home/bnu/venvs/deeplearning3.6/lib/python3.6/site-packages/tensorflow/python/framework/errors_impl.py", line 519, in __exit__
c_api.TF_GetCode(self.status.status))
tensorflow.python.framework.errors_impl.UnknownError: Fail to find the dnn implementation.
[[Node: rnn_1_2/CudnnRNN = CudnnRNN[T=DT_FLOAT, _class=["loc:@training/RMSprop/gradients/rnn_1_2/CudnnRNN_grad/CudnnRNNBackprop"], direction="unidirectional", dropout=0, input_mode="linear_input", is_training=true, rnn_mode="lstm", seed=87654321, seed2=0, _device="/job:localhost/replica:0/task:0/device:GPU:0"](rnn_1_2/transpose, rnn_1_2/ExpandDims_1, rnn_1_2/ExpandDims_1, rnn_1_2/concat)]]
[[Node: loss_2/mul/_241 = _Recv[client_terminated=false, recv_device="/job:localhost/replica:0/task:0/device:CPU:0", send_device="/job:localhost/replica:0/task:0/device:GPU:0", send_device_incarnation=1, tensor_name="edge_2251_loss_2/mul", tensor_type=DT_FLOAT, _device="/job:localhost/replica:0/task:0/device:CPU:0"]()]]
|
